546B - Soldier and Badges - CodeForces Solution


brute force greedy implementation sortings *1200

Please click on ads to support us..

Python Code:

import sys

native_input = input
input = lambda: sys.stdin.readline().strip()





import string

def solve():
    n, = map(int, input().split())
    a = list(map(int, input().split()))
    a.sort()

    l = []
    ans = 0

    for i in range(1, n):
        if a[i - 1] == a[i]:
            l.append(a[i])
        elif a[i - 1] + 1 < a[i]:
            for j in range(a[i - 1] + 1, a[i]):
                if not len(l):
                    break
                k = l.pop()
                ans += j - k

    if len(l):
        for i in range(len(l)):
            k = l.pop()
            ans += a[-1] + 1 + i - k

    print(ans)






solve()

C++ Code:

#include<bits/stdc++.h>
using namespace std;

#define endl "\n" ;
#define sort(v) sort(v.begin() , v.end());
#define reverse(v) reverse(v.begin() , v.end());
typedef pair<long long, long long > ii;
typedef vector<long long > vl;
typedef vector<int>vi ;
typedef vector<ii> vii;
typedef long long ll;

const int  N = 1e6 + 5;
int  oo = 100000;
ll const MOD = 1e9 + 7;


int dx[] = { 0, 1, -1,0 };
int dy[] = { 1,0 , 0,-1 };


int main() {

    ios_base::sync_with_stdio(false);
    cin.tie(NULL);



    int n ; cin >> n ;

    vi v(n);

    for (int i = 0 ; i < n ; i ++ )cin>>v[i];

    sort(v);
    int sol = 0 ;

    for (int i = 1 ; i < n ; i ++ ){

        for(int j = 0 ; j < i ; j ++ ){
            if(v[i]==v[j]){
                sol++;
                v[i]++;
            }
        }
    }

    cout <<sol;




    return 0;
}





Comments

Submit
0 Comments
More Questions

343B - Alternating Current
758B - Blown Garland
1681B - Card Trick
1592A - Gamer Hemose
493D - Vasya and Chess
1485A - Add and Divide
337B - Routine Problem
1392D - Omkar and Bed Wars
76E - Points
762C - Two strings
802M - April Fools' Problem (easy)
577B - Modulo Sum
1555B - Two Tables
1686A - Everything Everywhere All But One
1469B - Red and Blue
1257B - Magic Stick
18C - Stripe
1203B - Equal Rectangles
1536A - Omkar and Bad Story
1509A - Average Height
1506C - Double-ended Strings
340A - The Wall
377A - Maze
500A - New Year Transportation
908D - New Year and Arbitrary Arrangement
199A - Hexadecimal's theorem
519C - A and B and Team Training
631A - Interview
961B - Lecture Sleep
522A - Reposts